# Code for Table M2: Alternative cutpoints for pretrial incarceration measure

rm(list = ls())

## ---------------------------------------
## Load Packages 
## ---------------------------------------
require('AER')
require('ivpack')
require('dplyr')
require('data.table')

## ---------------------------------------
## Load Data and Functions
## ---------------------------------------
load("../data0812.RData")

directory <- "../functions/"
functions <- list.files(directory)  
loadfunctions <- sapply(functions, FUN = function(x)source(paste0(directory, x)))

## ---------------------------------------
## Two Stage Least Squares Specifications
## ---------------------------------------
time.controls <- "as.factor(court_time1) + as.factor(court_time2) + as.factor(court_dow) + as.factor(court_shift) + as.factor(totOGS2)"
case.controls <-   "as.factor(any_drug_2) +  as.factor(any_violent_2) + as.factor(fire_arms_2) +  as.factor(any_rob_2) + as.factor(any_dui_2) + as.factor(prior_offender_2)"
demo.controls <- "age_2012 + I(age_2012^2) + Female + as.factor(race) + vote2008 + as.factor(noteli08) + regis_before"

outc.1 <- "vote2012"
endo.1 <- "pti"
inst.1 <- "judgeiv"

form.1 <- formula(paste(outc.1, "~", endo.1, "+" , time.controls, "|", inst.1, "+", time.controls))
form.2 <- formula(paste(outc.1, "~", endo.1, "+" , time.controls, "+", demo.controls, "|", inst.1, "+", time.controls, "+", demo.controls))
form.3 <- formula(paste(outc.1, "~", endo.1, "+" , time.controls, "+", demo.controls, "+", case.controls, "|", inst.1, "+", time.controls, "+", demo.controls, "+", case.controls))

## ---------------------------------------
# Row 1: 5+ days
## ---------------------------------------
# define cutpoint
data0812$pti <- ifelse(data0812$days_jail < 5, 0, 1)
# construct instrument
data0812 <- constructIV(data0812)
# take the last case before the election
data0812 <- lastCase(data0812)


m1a1 <- ivreg(form.1, data = data0812)
m1a2 <- ivreg(form.2, data = data0812)
m1a3 <- ivreg(form.3, data = data0812)

m1a1_d <- summary(m1a1, vcov = sandwich, diagnostic = T)$diagnostics[1, 3]
m1a2_d <- summary(m1a2, vcov = sandwich, diagnostic = T)$diagnostics[1, 3]
m1a3_d <- summary(m1a3, vcov = sandwich, diagnostic = T)$diagnostics[1, 3]


row1 <- data.table(round(rbind(c(robust.se(m1a1)[2, c(1)], robust.se(m1a2)[2, c(1)], robust.se(m1a3)[2, c(1)]),
                               c(robust.se(m1a1)[2, c(2)], robust.se(m1a2)[2, c(2)], robust.se(m1a3)[2, c(2)]),
                               c(m1a1_d, m1a2_d, m1a3_d),
                               c(m1a1$n, m1a2$n, m1a3$n),
                               rep(NA, 3)
                               
),3))


## ---------------------------------------
# Row 2: 7+ days
## ---------------------------------------
# re-load data
load("../data0812.RData")

# define cutpoint
data0812$pti <- ifelse(data0812$days_jail < 7, 0, 1)

# construct instrument
data0812 <- constructIV(data0812)
# take the last case before the election
data0812 <- lastCase(data0812)

# re-run analyses
m1a1 <- ivreg(form.1, data = data0812)
m1a2 <- ivreg(form.2, data = data0812)
m1a3 <- ivreg(form.3, data = data0812)

m1a1_d <- summary(m1a1, vcov = sandwich, diagnostic = T)$diagnostics[1, 3]
m1a2_d <- summary(m1a2, vcov = sandwich, diagnostic = T)$diagnostics[1, 3]
m1a3_d <- summary(m1a3, vcov = sandwich, diagnostic = T)$diagnostics[1, 3]

row2 <- data.table(round(rbind(c(robust.se(m1a1)[2, c(1)], robust.se(m1a2)[2, c(1)], robust.se(m1a3)[2, c(1)]),
                               c(robust.se(m1a1)[2, c(2)], robust.se(m1a2)[2, c(2)], robust.se(m1a3)[2, c(2)]),
                               c(m1a1_d, m1a2_d, m1a3_d),
                               c(m1a1$n, m1a2$n, m1a3$n),
                               rep(NA, 3)
                               
),3))


## ---------------------------------------
# Row 3: 10+ days
## ---------------------------------------
# re-load data
load("../data0812.RData")

# define cutpoint
data0812$pti <- ifelse(data0812$days_jail < 10, 0, 1)

# construct instrument
data0812 <- constructIV(data0812)
# take the last case before the election
data0812 <- lastCase(data0812)

# re-run analyses
m1a1 <- ivreg(form.1, data = data0812)
m1a2 <- ivreg(form.2, data = data0812)
m1a3 <- ivreg(form.3, data = data0812)

m1a1_d <- summary(m1a1, vcov = sandwich, diagnostic = T)$diagnostics[1, 3]
m1a2_d <- summary(m1a2, vcov = sandwich, diagnostic = T)$diagnostics[1, 3]
m1a3_d <- summary(m1a3, vcov = sandwich, diagnostic = T)$diagnostics[1, 3]

row3 <- data.table(round(rbind(c(robust.se(m1a1)[2, c(1)], robust.se(m1a2)[2, c(1)], robust.se(m1a3)[2, c(1)]),
                               c(robust.se(m1a1)[2, c(2)], robust.se(m1a2)[2, c(2)], robust.se(m1a3)[2, c(2)]),
                               c(m1a1_d, m1a2_d, m1a3_d),
                               c(m1a1$n, m1a2$n, m1a3$n),
                               rep(NA, 3)
                               
),3))


## ---------------------------------------
# Row 4: 14+ days
## ---------------------------------------

# re-load data
load("../data0812.RData")

# define cutpoint
data0812$pti <- ifelse(data0812$days_jail <= 14, 0, 1)

# construct instrument
data0812 <- constructIV(data0812)
# take the last case before the election
data0812 <- lastCase(data0812)

# re-run analyses
m1a1 <- ivreg(form.1, data = data0812)
m1a2 <- ivreg(form.2, data = data0812)
m1a3 <- ivreg(form.3, data = data0812)

m1a1_d <- summary(m1a1, vcov = sandwich, diagnostic = T)$diagnostics[1, 3]
m1a2_d <- summary(m1a2, vcov = sandwich, diagnostic = T)$diagnostics[1, 3]
m1a3_d <- summary(m1a3, vcov = sandwich, diagnostic = T)$diagnostics[1, 3]

row4 <- data.table(round(rbind(c(robust.se(m1a1)[2, c(1)], robust.se(m1a2)[2, c(1)], robust.se(m1a3)[2, c(1)]),
                               c(robust.se(m1a1)[2, c(2)], robust.se(m1a2)[2, c(2)], robust.se(m1a3)[2, c(2)]),
                               c(m1a1_d, m1a2_d, m1a3_d),
                               c(m1a1$n, m1a2$n, m1a3$n),
                               rep(NA, 3)
),3))


## ---------------------------------------
# Row 5: continuous
## ---------------------------------------

# re-load data
load("../data0812.RData")

# define cutpoint
data0812$pti <- log1p(abs(data0812$days_jail))

# construct instrument
data0812 <- constructIV(data0812)
# take the last case before the election
data0812 <- lastCase(data0812)

# re-run analyses
m1a1 <- ivreg(form.1, data = data0812)
m1a2 <- ivreg(form.2, data = data0812)
m1a3 <- ivreg(form.3, data = data0812)

m1a1_d <- summary(m1a1, vcov = sandwich, diagnostic = T)$diagnostics[1, 3]
m1a2_d <- summary(m1a2, vcov = sandwich, diagnostic = T)$diagnostics[1, 3]
m1a3_d <- summary(m1a3, vcov = sandwich, diagnostic = T)$diagnostics[1, 3]

row5 <- data.table(round(rbind(c(robust.se(m1a1)[2, c(1)], robust.se(m1a2)[2, c(1)], robust.se(m1a3)[2, c(1)]),
                               c(robust.se(m1a1)[2, c(2)], robust.se(m1a2)[2, c(2)], robust.se(m1a3)[2, c(2)]),
                               c(m1a1_d, m1a2_d, m1a3_d),
                               c(m1a1$n, m1a2$n, m1a3$n)
),3))

## ---------------------------------------
# Table M2
## ---------------------------------------

cont <- rbind(row1, row2,  row3,  row4,row5)
colnames(cont) <- c("(1)", "(2)", "(3)")
# add row labels
cont$names <-  c("Pretrial Incarceration (PTI) - 5+",
                 "Std. Error",
                 "F-test", 
                 "Obs.",
                 "",
                 "Pretrial Incarceration (PTI) - 7+",
                 "Std. Error",
                 "F-test", 
                 "Obs.",
                 "",
                 "Pretrial Incarceration (PTI) - 10+",
                 "Std. Error",
                 "F-test", 
                 "Obs.", 
                 "",
                 "Pretrial Incarceration (PTI) - 14+",
                 "Std. Error",
                 "F-test", 
                 "Obs.",
                 "",
                 "Pretrial Incarceration (PTI) - continuous",
                 "Std. Error",
                 "F-test", 
                 "Obs.")
cont <- cont[, c("names", "(1)", "(2)", "(3)")]

cat("\nPrinting Table M2: Alternative Cutpoints...\n")
print(cont)





